package com.demo.generator;

import com.baomidou.mybatisplus.generator.InjectionConfig;
import com.baomidou.mybatisplus.generator.config.po.TableInfo;

import java.util.Map;

/**
 * @author gy
 * @date 2023/8/16
 */
public class MyInjectionConfig extends InjectionConfig {
    @Override
    public void initMap() {

    }

    /**
     * 依据表相关信息，从三方获取到需要元数据，处理方法环境里面
     *
     * @param tableInfo
     */
    @Override
    public void initTableMap(TableInfo tableInfo) {
        // 子类重写注入表对应补充信息
        String serviceName = tableInfo.getServiceName();
        // 修改Service接口名称 去除I前缀 为了适配框架代码
        tableInfo.setServiceName(serviceName.substring(1));
    }
}
